# pip install apache-nifi
'''
nifi-cli submit -template <template_path> -output-dir <output_directory>，其中<template_path>是模板文件的路径，<output_directory>是输出目录的路径。
'''

from nipyapi import config, nifi

config.host = 'http://localhost:8080' # Nifi服务器地址及端口号
config.username = 'admin' # Nifi管理员账户名称
config.password = 'admin' # Nifi管理员密码
current_flow = nifi.get_flow()
print(current_flow)
# 示例流程组JSON文件内容
pg_json = '''{
    "name": "MyFlow",
    "position": {
        "x": 0,
        "y": 0
    },
    ...
}'''
new_pg = nifi.create_process_group('root', pg_json)
print(new_pg)

# 完成流程设计后，可以使用nifi.update_flow()函数来部署修改后的流程
updated_flow = nifi.update_flow(current_flow['id'], current_flow)
print(updated_flow)
